from django.contrib import admin
from brp_app.models import Book, TOCEntry



class BookAdmin(admin.ModelAdmin):
    list_display = ('title', 'pages', 'toc_file', 'date_added', 'date_started', 'date_finished', 'status', 'max_level')



class TOCEntryAdmin(admin.ModelAdmin):
    list_display = ('book', 'page', 'title', 'level', 'read')
    list_display_links = ('title', )


admin.site.register(Book, BookAdmin)
admin.site.register(TOCEntry, TOCEntryAdmin)
